package com.hui.qiniucloud.dto;

import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;

import java.util.List;

/**
 * 文生图响应DTO
 */
@Data
@Schema(description = "文生图响应")
public class ImageGenerateResponse {

    @Schema(description = "使用的模型名称", example = "doubao-seedream-3-0-t2i-250415")
    private String model;

    @Schema(description = "创建时间戳", example = "1589478378")
    private Long created;

    @Schema(description = "生成的图像数据列表")
    private List<ImageData> data;

    @Schema(description = "使用情况统计")
    private Usage usage;

    @Data
    @Schema(description = "图像数据")
    public static class ImageData {
        @Schema(description = "图像URL", example = "https://...")
        private String url;

        @Schema(description = "图像的base64编码数据")
        private String b64Json;
    }

    @Data
    @Schema(description = "使用情况统计")
    public static class Usage {
        @Schema(description = "生成的图像数量", example = "1")
        private Integer generatedImages;

        @Schema(description = "输出token数", example = "100")
        private Integer outputTokens;

        @Schema(description = "总token数", example = "150")
        private Integer totalTokens;
    }
}